#!/usr/bin/env python

# 这是一个修改版的 learner.py 文件，修改了 use_threads 函数，使其能够处理没有 concurrency 字段的策略配置

# 导入原始 learner.py 中的所有内容
from lerobot.scripts.rl.learner import *

# 重写 use_threads 函数
def use_threads(cfg: TrainPipelineConfig) -> bool:
    try:
        return cfg.policy.concurrency.learner == "threads"
    except AttributeError:
        # 如果策略配置没有 concurrency 字段，默认使用线程
        return True

# 如果直接运行此脚本
if __name__ == "__main__":
    train_cli()
    logging.info("[LEARNER] main finished")
